/*
 * Funambol is a mobile platform developed by Funambol, Inc. 
 * Copyright (C) 2008 Funambol, Inc.
 * 
 * This program is free software; you can redistribute it and/or modify it under
 * the terms of the GNU Affero General Public License version 3 as published by
 * the Free Software Foundation with the addition of the following permission 
 * added to Section 15 as permitted in Section 7(a): FOR ANY PART OF THE COVERED
 * WORK IN WHICH THE COPYRIGHT IS OWNED BY FUNAMBOL, FUNAMBOL DISCLAIMS THE 
 * WARRANTY OF NON INFRINGEMENT  OF THIRD PARTY RIGHTS.
 * 
 * This program is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
 * details.
 * 
 * You should have received a copy of the GNU Affero General Public License 
 * along with this program; if not, see http://www.gnu.org/licenses or write to
 * the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 * MA 02110-1301 USA.
 * 
 * You can contact Funambol, Inc. headquarters at 643 Bair Island Road, Suite 
 * 305, Redwood City, CA 94063, USA, or at email address info@funambol.com.
 * 
 * The interactive user interfaces in modified source and object code versions
 * of this program must display Appropriate Legal Notices, as required under
 * Section 5 of the GNU Affero General Public License version 3.
 * 
 * In accordance with Section 7(b) of the GNU Affero General Public License
 * version 3, these Appropriate Legal Notices must retain the display of the
 * "Powered by Funambol" logo. If the display of the logo is not reasonably 
 * feasible for technical reasons, the Appropriate Legal Notices must display
 * the words "Powered by Funambol".
 */
 
#include <sqlite3.h>
#include "SQLiteDriver.h"
#include "base/Log.h"

SQLiteDriver::SQLiteDriver(const char* cpath){
    path = cpath;
    db = NULL;
    statement = NULL;
    connect();
}


SQLiteDriver::~SQLiteDriver(){
    if (statement)
    {
        sqlite3_finalize(statement);
        statement = NULL;
    }
    disconnect();
}


int SQLiteDriver::connect(){
    if (db) {        
        return SQLITE_OK;
    } 
    
    int ret = sqlite3_open(path, &db);
    if (ret == SQLITE_OK) {
        LOG.debug("SQLiteDriver connect");
        return execute("BEGIN TRANSACTION;");
    }
    return ret;
}

int SQLiteDriver::execute(const char* sql){
    connect();
    
    int ret = sqlite3_prepare(db, sql, strlen(sql), &statement, NULL);
    LOG.debug("execute query: %s", sql);
    if (ret == SQLITE_OK) {

        ret = sqlite3_step(statement);
    } /*else {
        LOG.error("Unable to prepare sqlite query in execute(): %s", sql.c_str());
    }*/
    if (ret != SQLITE_OK && ret != SQLITE_DONE) {
        LOG.error("Error executing in the db %i", ret);
        //LOG.error("Error executing in the db: %s", sql.c_str());
        return ret;
    }
    
    ret = sqlite3_finalize(statement);
    statement = NULL;
    if (ret == SQLITE_DONE) {
        ret = SQLITE_OK;
    }
    /*if (ret != SQLITE_OK) {
        LOG.error("SQLite execute failed: %s", sqlite3_errmsg(db));
    }*/
    return ret;
}

int SQLiteDriver::disconnect(){
    if (db == NULL)
        return SQLITE_OK;

    execute("ROLLBACK TRANSACTION;");
        LOG.debug("SQLiteDriver disconnect");
    int ret = sqlite3_close(db);
    db = NULL;
    return ret;
}

int SQLiteDriver::queryIndex(const char* query, ArrayList& result) {
    
    connect();
    LOG.debug("query index connected to debug");
    
    if (statement) {
        sqlite3_finalize(statement);
        statement = NULL;
    }    
    //
    // The follow query locks the db until a sqlite3_finalize call
    //int ret = sqlite3_prepare(db, query, strlen(query), &statement, NULL);    
    LOG.debug("queryIndex: %s", query);        
    
    char ** retStrings = 0;
    char * errmsg = 0;
    int iRows = 0, iCols = 0;
    int iErr = sqlite3_get_table(db, query, &retStrings, &iRows, &iCols, &errmsg);
	
	LOG.debug("The rows are %i", iRows);
	LOG.debug("The column are %i", iCols);
	
    // the 0 value is the name of the column...
    int i = 0;
    for (i = 1; i <= iRows; i++) {        
        StringBuffer s(retStrings[i]);
        result.add(s);                
    }           

    sqlite3_free_table(retStrings);

    return 0;   
}

int SQLiteDriver::commit(){
    if (statement) {
        sqlite3_finalize(statement);
        statement = NULL;
    }
    
    return (
        ((execute("COMMIT TRANSACTION;") == SQLITE_OK) && 
         (execute("BEGIN TRANSACTION;") == SQLITE_OK)) 
          ? 0 : 1);
}

int SQLiteDriver::queryColumns(const char* query, ArrayList& result) {
    
    connect();
    LOG.debug("queryColumns connected to debug");
    
    if (statement) {
        sqlite3_finalize(statement);
        statement = NULL;
    }    
    //
    // The follow query locks the db until a sqlite3_finalize call
    //int ret = sqlite3_prepare(db, query, strlen(query), &statement, NULL);    
    LOG.debug("queryColumns: %s", query);        
    
    char ** retStrings = 0;
    char * errmsg = 0;
    int iRows = 0, iCols = 0;
    int iErr = sqlite3_get_table(db, query, &retStrings, &iRows, &iCols, &errmsg);

	LOG.debug("The queryColumns rows are %i", iRows);
	LOG.debug("The queryColumns column are %i", iCols);
	if (iRows > 0) {	
		int itotal = (iRows + 1) * iCols;
		// the 0 value is the name of the column...
		int i = 0;
		for (i = iCols; i < itotal; i++) {        
			StringBuffer s(retStrings[i]);
			LOG.debug("value = %s", s.c_str());
			result.add(s);                
		}           
	}
    sqlite3_free_table(retStrings);

    return 0;   
}


